@echo off
chcp 65001 >nul
echo HomeBox 数据库导入重启脚本
echo ================================
echo.

echo [1/3] 检查重启标记文件...
if not exist "RESTART_REQUIRED_DB" (
    echo ❌ 未找到数据库导入标记文件
    echo 请先通过工具页面上传数据库文件
    echo.
    pause
    exit /b 1
)

echo ✓ 找到数据库导入标记文件

echo.
echo [2/3] 读取临时数据库路径...
set /p TEMP_DB_PATH=<RESTART_REQUIRED_DB
echo 临时数据库路径: %TEMP_DB_PATH%

if not exist "%TEMP_DB_PATH%" (
    echo ❌ 临时数据库文件不存在: %TEMP_DB_PATH%
    del RESTART_REQUIRED_DB 2>nul
    pause
    exit /b 1
)

echo.
echo [3/3] 开始数据库替换过程...

echo 停止现有服务...
taskkill /F /IM homebox.exe 2>nul
taskkill /F /IM homebox-enhanced.exe 2>nul
taskkill /F /IM main.exe 2>nul
timeout /t 2 >nul

echo 替换数据库文件...
if exist "data\homebox.db" (
    echo 备份当前数据库为 data\homebox.db.backup
    copy "data\homebox.db" "data\homebox.db.backup" >nul
    if errorlevel 1 (
        echo ❌ 备份当前数据库失败
        pause
        exit /b 1
    )
)

echo 使用新数据库文件...
copy "%TEMP_DB_PATH%" "data\homebox.db" >nul
if errorlevel 1 (
    echo ❌ 复制新数据库失败
    pause
    exit /b 1
)

echo 清理临时文件...
del "%TEMP_DB_PATH%" 2>nul
del RESTART_REQUIRED_DB 2>nul

echo.
echo ✅ 数据库导入完成！
echo.
echo 现在可以启动 HomeBox 服务:
echo   1. 运行 run-backend.bat 启动后端服务
echo   2. 运行 run-frontend.bat 启动前端服务
echo.
echo 或使用 start-all.bat 同时启动前后端
echo.
pause